package leetcode._07_贪心;

import org.junit.Test;

import java.util.ArrayList;
import java.util.HashMap;
import java.util.LinkedList;
import java.util.PriorityQueue;

/**
 * @author pppppp
 * @date 2022/3/17 16:10
 * 给你一个非负整数数组 nums ，你最初位于数组的第一个位置。
 * 数组中的每个元素代表你在该位置可以跳跃的最大长度。
 * 你的目标是使用最少的跳跃次数到达数组的最后一个位置。
 * 假设你总是可以到达数组的最后一个位置。
 * 示例 1:
 * 输入: nums = [2,3,1,1,4]
 * 输出: 2
 * 解释: 跳到最后一个位置的最小跳跃数是 2。
 *      从下标为 0 跳到下标为 1 的位置，跳 1 步，然后跳 3 步到达数组的最后一个位置。
 * <p>
 * 示例 2:
 * 输入: nums = [2,3,0,1,4]
 * 输出: 2
 * 提示:
 * <p>
 * 1 <= nums.length <= 104
 * 0 <= nums[i] <= 1000。
 */
public class _45_跳跃游戏II {

    @Test
    public void T_0() {
        int[][] nums = {{1, 1, 1, 1}, {1, 2}, {2, 3, 1, 1, 4}, {2, 3, 0, 1, 4}};
        int[] ans = {3, 1, 2, 2};
        for (int i = 0; i < nums.length; i++) {
            // System.out.println(jump_0(nums[i]) == ans[i]);
            // System.out.println(jump_G2_yh(nums[i]) == ans[i]);
            System.out.println(jump_G1(nums[i]) == ans[i]);
        }
    }

    /*1.官方题解--正向解答*/
    public int jump_G1(int[] nums) {
        int length = nums.length;
        int end = 0;
        int maxPosition = 0;
        int steps = 0;
        for (int i = 0; i < length - 1; i++) {
            maxPosition = Math.max(maxPosition, i + nums[i]);
            if(maxPosition >= nums.length-1){
                return ++steps;
            }
            if (i == end) {
                end = maxPosition;
                steps++;
            }
        }
        return steps;
    }

    /*2.逆向题解*/
    public int jump_G2_yh(int[] nums) {
        HashMap<Integer, PriorityQueue<Integer>> map = new HashMap<>();
        for (int i = 0; i < nums.length; i++) {
            PriorityQueue<Integer> queue = map.getOrDefault(Math.min(nums[i] + i,nums.length-1), new PriorityQueue<>());
            queue.add(i);
            map.put(Math.min(nums[i] + i,nums.length-1),queue);
        }
        int position = nums.length - 1; //要找的位置
        int steps = 0;
        while (position != 0) { //是否到了第 0 个位置
            PriorityQueue<Integer> queue = map.get(position);
            position =  queue.peek();
            steps++;
        }
        return steps;
    }

    public int jump_G2(int[] nums) {
        int position = nums.length - 1; //要找的位置
        int steps = 0;
        while (position != 0) { //是否到了第 0 个位置
            for (int i = 0; i < position; i++) {
                /*从左往右找到第一个能到达的点*/
                if (nums[i] + i>= position) {
                    position = i; //更新要找的位置
                    steps++;
                    break;
                }
            }
        }
        return steps;
    }


    public int jump_0(int[] nums) {
        if (nums.length == 1) {
            return 0;
        }
        int step = 0, maxIndex = 0;
        for (int i = 0; i < nums.length; i++) {
            int nextIndex = nums[i] + i;
            if (nextIndex >= nums.length - 1) {
                step++;
                break;
            }
            for (int j = i + 1; j <= nextIndex; j++) {
                maxIndex = Math.max(maxIndex, nums[j] + j);
            }
            if (maxIndex >= nums.length - 1) {
                step++;
                break;
            }
            i = maxIndex - 1;
            step++;
        }
        return step;
    }


}
